package com.yaolin.generator.api.enjoy.mapper;

import java.util.ArrayList;
import java.util.List;

import com.yaolin.generator.api.Column;
import com.yaolin.generator.api.Table;

public class Mapper {

	public static TableDTO mapToDTO(Table table) {
		TableDTO dto = new TableDTO();
		
		dto.setName(table.getName());
		dto.setRemark(table.getRemark());
		dto.setListColumnDTO(mapToDTO(table.getListColumn()));
		
		return dto;
	}
	
	public static List<ColumnDTO> mapToDTO(List<Column> list) {
		List<ColumnDTO> listResult = new ArrayList<>();
		if (list != null) {
			for (Column column : list) {
				ColumnDTO dto = new ColumnDTO();
				dto.setName(column.getName());
				dto.setClassName(column.getClassName());
				listResult.add(dto);
			}
		}
		return listResult;
	}
}
